package com.foosun.util.wechat;

import java.io.InputStream;
import java.io.Writer;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;

import org.dom4j.Document;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;
import org.w3c.dom.NodeList;

import com.foosun.entity.message.Article;
import com.foosun.entity.message.res.MusicMessage;
import com.foosun.entity.message.res.NewsMessage;
import com.foosun.entity.message.res.TextMessage;
import com.foosun.entity.message.res.TransferCustomerServiceMessage;
import com.thoughtworks.xstream.XStream;
import com.thoughtworks.xstream.core.util.QuickWriter;
import com.thoughtworks.xstream.io.HierarchicalStreamWriter;
import com.thoughtworks.xstream.io.xml.PrettyPrintWriter;
import com.thoughtworks.xstream.io.xml.XppDriver;

/**
 * 消息工具类
 * 
 * @author leo
 * @date 2013-12-18
 */
public class MessageUtil {

	/**
	 * 返回消息类型：文本
	 */
	public static final String RESP_MESSAGE_TYPE_TEXT = "text";

	/**
	 * 请求消息类型：多客服消息
	 */
	public static final String RESP_MESSAGE_TYPE_TRANSFER = "transfer_customer_service";
	
	/**
	 * 返回消息类型：音乐
	 */
	public static final String RESP_MESSAGE_TYPE_MUSIC = "music";

	/**
	 * 返回消息类型：图文
	 */
	public static final String RESP_MESSAGE_TYPE_NEWS = "news";

	/**
	 * 请求消息类型：文本
	 */
	public static final String REQ_MESSAGE_TYPE_TEXT = "text";

	/**
	 * 请求消息类型：图片
	 */
	public static final String REQ_MESSAGE_TYPE_IMAGE = "image";

	/**
	 * 请求消息类型：链接
	 */
	public static final String REQ_MESSAGE_TYPE_LINK = "link";

	/**
	 * 请求消息类型：地理位置
	 */
	public static final String REQ_MESSAGE_TYPE_LOCATION = "location";

	/**
	 * 请求消息类型：音频
	 */
	public static final String REQ_MESSAGE_TYPE_VOICE = "voice";

	/**
	 * 请求消息类型：推送
	 */
	public static final String REQ_MESSAGE_TYPE_EVENT = "event";

	/**
	 * 事件类型：subscribe(订阅)
	 */
	public static final String EVENT_TYPE_SUBSCRIBE = "subscribe";

	/**
	 * 事件类型：unsubscribe(取消订阅)
	 */
	public static final String EVENT_TYPE_UNSUBSCRIBE = "unsubscribe";

	/**
	 * 事件类型：CLICK(自定义菜单点击事件)
	 */
	public static final String EVENT_TYPE_CLICK = "CLICK";
	
	/**
	 * 事件类型：VIEW(自定义菜单查看事件)
	 */
	public static final String EVENT_TYPE_VIEW = "VIEW";

	/**
	 * 事件主题：关于GOGO(自定义菜单点击参数)
	 */
	public static final String KEY_ABOUT_MMYX = "V1001_ABOUT_MMYX";
	
	/**
	 * 事件主题：赞一下我们(自定义菜单点击参数)
	 */
	public static final String KEY_GOOD = "V1001_GOOD";
	
	/**
	 * 事件主题：赞一下我们(自定义菜单点击参数)
	 */
	public static final String KEY_CUSTOMER_SERVICE = "V1001_CUSTOMER_SERVICE";
	
	

	/**
	 * 解析微信发来的请求（XML）
	 * 
	 * @param request
	 * @return
	 * @throws Exception
	 */
	@SuppressWarnings("unchecked")
	public static Map<String, String> parseXml(HttpServletRequest request) throws Exception {
		// 将解析结果存储在HashMap中
		Map<String, String> map = new HashMap<String, String>();

		// 从request中取得输入流
		InputStream inputStream = request.getInputStream();
		// 读取输入流
		SAXReader reader = new SAXReader();
		Document document = reader.read(inputStream);
		// 得到xml根元素
		Element root = document.getRootElement();
		// 得到根元素的所有子节点
		List<Element> elementList = root.elements();

		// 遍历所有子节点
		for (Element e : elementList)
			map.put(e.getName(), e.getText());

		// 释放资源
		inputStream.close();
		inputStream = null;

		return map;
	}
	
	/**
	 * 解析微信发来的请求（XML）
	 * 
	 * @param request
	 * @return
	 * @throws Exception
	 */
	@SuppressWarnings("unchecked")
	public static Map<String, String> parseXml(String secret) throws Exception {
		// 将解析结果存储在HashMap中
		Map<String, String> map = new HashMap<String, String>();

		// 解密
		String result = SignUtil.decrypt(secret);
		System.out.println("after decrypt msg: " + result);
		Document document = DocumentHelper.parseText(result);
		
		// 得到xml根元素
		Element root = document.getRootElement();
		
		// 得到根元素的所有子节点
		List<Element> elementList = root.elements();

		// 遍历所有子节点
		for (Element e : elementList)
			map.put(e.getName(), e.getText());

		return map;
	}

	/**
	 * 文本消息对象转换成xml
	 * 
	 * @param textMessage 文本消息对象
	 * @return xml
	 */
	public static String textMessageToXml(TextMessage textMessage) {
		xstream.alias("xml", textMessage.getClass());
		return xstream.toXML(textMessage);
	}

	/**
	 * 音乐消息对象转换成xml
	 * 
	 * @param musicMessage 音乐消息对象
	 * @return xml
	 */
	public static String musicMessageToXml(MusicMessage musicMessage) {
		xstream.alias("xml", musicMessage.getClass());
		return xstream.toXML(musicMessage);
	}

	/**
	 * 图文消息对象转换成xml
	 * 
	 * @param newsMessage 图文消息对象
	 * @return xml
	 */
	public static String newsMessageToXml(NewsMessage newsMessage) {
		xstream.alias("xml", newsMessage.getClass());
		xstream.alias("item", new Article().getClass());
		return xstream.toXML(newsMessage);
	}
	
	/**
	 * 多客服对象转换成xml
	 * 
	 * @param transferCustomerServiceMessage 多客服信号对象
	 * @return xml
	 */
	public static String transferCustomerServiceMessageToXml(TransferCustomerServiceMessage transferCustomerServiceMessage) {
		xstream.alias("xml", transferCustomerServiceMessage.getClass());
		return xstream.toXML(transferCustomerServiceMessage);
	}
	
	/**
	 * 扩展xstream，使其支持CDATA块
	 * 
	 * @date 2013-05-19
	 */
	private static XStream xstream = new XStream(new XppDriver() {
		public HierarchicalStreamWriter createWriter(Writer out) {
			return new PrettyPrintWriter(out) {
				// 对所有xml节点的转换都增加CDATA标记
				boolean cdata = true;

				public void startNode(String name, @SuppressWarnings("rawtypes") Class clazz) {
					super.startNode(name, clazz);
				}

				protected void writeText(QuickWriter writer, String text) {
					if (cdata) {
						writer.write("<![CDATA[");
						writer.write(text);
						writer.write("]]>");
					} else {
						writer.write(text);
					}
				}
			};
		}
	});
	
	
	/**
	 * 解析获取到的用户信息字符串
	 * @param user
	 * @return
	 * @throws Exception
	 */
	public static Map<String, String> parseUserinfo(String user) throws Exception {
		// 将解析结果存储在HashMap中
		Map<String, String> map = new HashMap<String, String>();
		user = user.substring(user.indexOf("{")+1,user.indexOf("}"));
		user = user.replaceAll("\"", "");
		String[] temp = user.split(",");
		for(int i =0;i<temp.length;i++){
			String[] t = temp[i].split(":");
			if(t.length == 1)
				map.put(t[0], "");
			else if(t.length == 3)
				map.put(t[0], t[1]+":"+t[2]);
			else
				map.put(t[0], t[1]);
		}
		return map;
	}
	
	/**
	 * 解析openid列表
	 * @param openidlist
	 * @return
	 * @throws Exception
	 */
	public static String[] parseOpenidList(String openidlist) throws Exception {
		openidlist = openidlist.substring(openidlist.indexOf("[")+1,openidlist.indexOf("]"));
		openidlist = openidlist.replaceAll("\"", "");
		String[] temp = openidlist.split(","); 		
		return temp;
	}
}

